BiPACE: Bisimulation-Guided Policy Optimization with Action Counterfactual Estimation for LLM Agents
BiPACE:基于双模拟引导与动作反事实估计的LLM智能体策略优化
🧭 LLM Agent 训练 · 分组策略优化
University of Chicago · Stanford University · The Hong Kong University of Science and Technology (Guangzhou) · University of Science and Technology of China · Meituan
用双模拟聚类和动作反事实基线修复步进分组RL的状态-动作信用失配,将ALFWorld验证成功率从90.8%提升至97.1%
📌 工作参考价值(4 点)
- 在训练对话式推荐、搜索 Agent 等多步决策模型时,可借鉴 BiGPO:使用 Actor 自身隐藏状态的余弦距离在线聚类,替代观测哈希,作为状态近似双模拟关系,大幅减少孤立组,增强组内信号,适用于并行采样多条轨迹的在线训练。
- PACE 的动作反事实基线:在行为簇内用动作条件化的均值作为基线计算 Q(s,a)-V(s) 风格的优势,避免将状态价值误归因给动作,适用于推荐 Agent 中相似状态下不同动作(如提问、推品)的精确信用分配。
- 方法无额外 Critic 网络,只需存储 rollout 中的隐藏状态和分组信息,额外开销仅 11.3% 训练步长,工程实现轻量,可直接替换现有 stepwise GRPO/GiGPO 的优势估计器,提升长程任务成功率。
- 限制:依赖多条并行轨迹构建分组,若系统仅支持单轨迹在线学习,需做适配;但批量离线训练或仿真交互场景可直接受益。